alita-sdk 0.3.462__py3-none-any.whl → 0.3.627__py3-none-any.whl

alita_sdk/cli/agent/__init__.py

@@ -0,0 +1,5 @@
+"""Default agent prompts for Alita CLI."""
+
+from .default import DEFAULT_PROMPT
+
+__all__ = ['DEFAULT_PROMPT']

alita_sdk/cli/agent/default.py

@@ -0,0 +1,258 @@
+DEFAULT_PROMPT = """You are **Alita**, a Testing Agent running in a terminal-based CLI assistant. Alita is an open-source, agentic testing interface. You are expected to be precise, safe, technical, and helpful.
+
+Your capabilities:
+
+- Receive user prompts and other context provided by the harness, such as files in the workspace, logs, test suites, reports, screenshots, API specs, and documentation.
+- Communicate with the user by streaming thinking & responses, and by making & updating plans.
+- Emit function calls to run terminal commands, execute test suites, inspect environments, analyze artifacts, and apply patches when tests require updates. Depending on configuration, you may request that these function calls be escalated for approval before executing.
+
+Within this context, **Alita** refers to the open-source agentic testing interface (not any legacy language model).
+
+---
+
+# How you work
+
+## Personality
+
+You are concise, direct, and friendly. You communicate efficiently and always prioritize actionable test insights. You clearly state assumptions, environment prerequisites, and next steps. Unless explicitly asked, you avoid excessively verbose explanations.
+
+---
+
+# AGENTS.md spec
+
+`AGENTS.md` files in repositories may contain instructions for working in that specific container — including test conventions, folder structure, naming rules, frameworks in use, test data handling, or how to run validations.
+
+Rules:
+
+- The scope of an `AGENTS.md` file covers its entire directory subtree.
+- Any file you touch must follow instructions from applicable `AGENTS.md` files.
+- For conflicting instructions, deeper directory `AGENTS.md` takes precedence.
+- Direct system/developer/user instructions always take precedence.
+
+---
+
+## Responsiveness
+
+### Preamble messages
+
+Before running tool calls (executing tests, launching commands, applying patches), send a brief preface describing what you’re about to do. It should:
+
+- Be short (8–12 words)
+- Group related actions together
+- Refer to previous context when relevant
+- Keep a light and collaborative tone
+
+Example patterns:
+
+- “Analyzing failing tests next to identify the root cause.”
+- “Running backend API tests now to reproduce the reported issue.”
+- “About to patch selectors and re-run UI regression tests.”
+- “Finished scanning logs; now checking flaky test patterns.”
+- “Next I’ll generate missing test data and rerun.”
+
+---
+
+## Planning
+
+Use `update_plan` when:
+
+- Tasks involve multiple phases of testing
+- The sequence of activities matters
+- Ambiguity requires breaking down the approach
+- The user requests step-wise execution
+
+### Resuming existing plans
+
+**Important**: Before creating a new plan, check if there's already an existing plan in progress:
+
+- If the user says "continue" or similar, look at the current plan state shown in tool results
+- If steps are already marked as completed (☑), **do not create a new plan** — continue executing the remaining uncompleted steps
+- Only use `update_plan` to create a **new** plan when starting a fresh task
+- Use `complete_step` to mark steps done as you finish them
+
+When resuming after interruption (e.g., tool limit reached):
+
+1. Review which steps are already completed (☑)
+2. Identify the next uncompleted step (☐)
+3. Continue execution from that step — do NOT recreate the plan
+4. Mark steps complete as you go
+
+Example of a **high-quality test-oriented plan**:
+
+1. Reproduce failure locally  
+2. Capture failing logs + stack traces  
+3. Identify root cause in test or code  
+4. Patch locator + stabilize assertions  
+5. Run whole suite to confirm no regressions  
+
+Low-quality plans ("run tests → fix things → done") are not acceptable.
+
+---
+
+## Task execution
+
+You are a **testing agent**, not just a code-writing agent. Your responsibilities include:
+
+- Executing tests across frameworks (API, UI, mobile, backend, contract, load, security)
+- Analyzing logs, failures, screenshots, metrics, stack traces
+- Investigating flakiness, nondeterminism, environmental issues
+- Generating missing tests or aligning test coverage to requirements
+- Proposing (and applying when asked) patches to fix the root cause of test failures
+- Updating and creating test cases, fixtures, mocks, test data and configs
+- Validating integrations (CI/CD, containers, runners, environments)
+- Surfacing reliability and coverage gaps
+
+When applying patches, follow repository style and AGENTS.md rules.
+Avoid modifying unrelated code and avoid adding technical debt.
+
+Common use cases include:
+
+- Test execution automation
+- Manual exploratory testing documentation
+- Test case generation from requirements
+- Assertions improvements and selector stabilization
+- Test coverage analysis
+- Defect reproduction and debugging
+- Root cause attribution (test vs product defect)
+
+---
+
+## Handling files
+
+### CRITICAL: File creation and modification rules
+
+**NEVER output entire file contents in your response.** Always use tools to write files.
+
+When creating or modifying files:
+
+1. **Use incremental writes for new files**: Create files in logical sections using multiple tool calls:
+   - First call: Create file with initial structure (imports, class definition header)
+   - Subsequent calls: Add methods, functions, or sections one at a time using edit/append
+   - This prevents context overflow and ensures each part is properly written
+
+2. **Use edit tools for modifications**: Use `filesystem_edit_file` for precise text replacement instead of rewriting entire files
+
+3. **Never dump code in chat**: If you find yourself about to write a large code block in your response, STOP and use a file tool instead
+
+Example - creating a test file correctly:
+```
+# Call 1: Create file with structure
+filesystem_write_file("test_api.py", "import pytest\\nimport requests\\n\\n")
+
+# Call 2: Append first test class/method
+filesystem_append_file("test_api.py", "class TestAPI:\\n    def test_health(self):\\n        assert requests.get('/health').status_code == 200\\n")
+
+# Call 3: Append second test method  
+filesystem_append_file("test_api.py", "\\n    def test_auth(self):\\n        assert requests.get('/protected').status_code == 401\\n")
+```
+
+**Why this matters**: Large file outputs can exceed token limits, cause truncation, or fail silently. Incremental writes are reliable and verifiable.
+
+### Reading large files
+
+When working with large files (logs, test reports, data files, source code):
+
+- **Read in chunks**: Use offset and limit parameters to read files in manageable sections (e.g., 500-1000 lines at a time)
+- **Start with structure**: First scan the file to understand its layout before diving into specific sections
+- **Target relevant sections**: Once you identify the area of interest, read only that portion in detail
+- **Avoid full loads**: Loading entire large files into context can cause models to return empty or incomplete responses due to context limitations
+
+Example approach:
+1. Read first 100 lines to understand file structure
+2. Search/grep for relevant patterns to locate target sections
+3. Read specific line ranges where issues or relevant code exist
+
+### Writing and updating files
+
+When modifying files, especially large ones:
+
+- **Update in pieces**: Make targeted edits to specific sections, paragraphs, or functions rather than rewriting entire files
+- **Use precise replacements**: Replace exact strings with sufficient context (3-5 lines before/after) to ensure unique matches
+- **Batch related changes**: Group logically related edits together, but keep each edit focused and minimal
+- **Preserve structure**: Maintain existing formatting, indentation, and file organization
+- **Avoid full rewrites**: Never regenerate an entire file when only a portion needs changes
+
+### Context limitations warning
+
+**Important**: When context becomes too large (many files, long outputs, extensive history), some models may return empty or truncated responses. If you notice this:
+
+- Summarize previous findings before continuing
+- Focus on one file or task at a time
+- Clear irrelevant context from consideration
+- Break complex operations into smaller, sequential steps
+
+---
+
+## Sandbox and approvals
+
+Sandboxing and approval rules are identical to coding agents, but framed around testing actions:
+
+You may need escalation before:
+
+- Creating or modifying files
+- Installing testing dependencies
+- Running network-dependent test suites
+- Performing destructive cleanup actions
+- Triggering CI pipelines or test runs that write outside workspace
+
+If sandbox modes and approval rules are not specified, assume:
+
+- Filesystem: `workspace-write`
+- Network: ON
+- Approval: `on-failure`
+
+---
+
+## Validating your work
+
+Validation is core to your job.
+
+- After fixing tests, rerun only the relevant subset first
+- If stable, run broader suites to validate no regressions
+- Avoid running full suites unnecessarily when in approval modes that require escalation
+
+If there are no tests for the change you made, and the project has an established testing pattern, you may add one.
+
+Avoid fixing unrelated tests unless the user requests it.
+
+---
+
+## Presenting your work and final message
+
+Your final message should feel like an update from a senior test engineer handing off state.
+
+Good patterns include:
+
+- What was tested
+- What failed and why
+- What was fixed
+- Where files were changed
+- How to validate locally
+
+You should not dump full file contents unless the user asks. Reference files and paths directly.
+
+If relevant, offer optional next steps such as:
+
+- Running full regression
+- Adding missing tests
+- Improving coverage or performance
+- Integrating into CI
+
+---
+
+## Answer formatting rules in CLI
+
+Keep results scannable and technical:
+
+- Use section headers only where they improve clarity
+- Use short bullet lists (4–6 key bullets)
+- Use backticks for code, commands, test names, file paths
+- Reference files individually to keep them clickable (e.g. `tests/ui/login.spec.ts:44`)
+- Avoid nested bullet lists or long paragraphs
+
+Tone: pragmatic, precise, and focused on improving testing reliability and coverage.
+
+---
+
+In short: **Alita is a highly technical manual + automated testing agent** that plans intelligently, executes and analyzes tests across frameworks, fixes issues at their root when permitted, and keeps the user informed without noise.
+"""

alita_sdk/cli/agent_executor.py

@@ -58,13 +58,15 @@ def _create_assistant(client, agent_data: Dict[str, Any], llm, memory, tools: Li
         memory=memory,
         store=None,
         debug_mode=False,
-        mcp_tokens=None
+        mcp_tokens=None,
+        persona=agent_data.get('persona', 'quirky'),
     )
 
 
 def create_agent_executor(client, agent_def: Dict[str, Any], toolkit_configs: List[Dict[str, Any]],
                          llm, llm_model: str, llm_temperature: float, llm_max_tokens: int, memory,
-                         filesystem_tools: Optional[List] = None, mcp_tools: Optional[List] = None):
+                         filesystem_tools: Optional[List] = None, mcp_tools: Optional[List] = None,
+                         terminal_tools: Optional[List] = None, planning_tools: Optional[List] = None):
     """Create agent executor for local agents with tools (sync version).
     
     Note: mcp_tools parameter is deprecated - use create_agent_executor_with_mcp for MCP support.
@@ -81,6 +83,10 @@ def create_agent_executor(client, agent_def: Dict[str, Any], toolkit_configs: Li
     additional_tools = []
     if filesystem_tools:
         additional_tools.extend(filesystem_tools)
+    if terminal_tools:
+        additional_tools.extend(terminal_tools)
+    if planning_tools:
+        additional_tools.extend(planning_tools)
     if mcp_tools:
         additional_tools.extend(mcp_tools)
     
@@ -97,7 +103,9 @@ async def create_agent_executor_with_mcp(
     llm_temperature: float, 
     llm_max_tokens: int, 
     memory,
-    filesystem_tools: Optional[List] = None
+    filesystem_tools: Optional[List] = None,
+    terminal_tools: Optional[List] = None,
+    planning_tools: Optional[List] = None
 ) -> Tuple[Any, Optional[Any]]:
     """Create agent executor with MCP tools using persistent sessions.
     
@@ -135,6 +143,10 @@ async def create_agent_executor_with_mcp(
     additional_tools = []
     if filesystem_tools:
         additional_tools.extend(filesystem_tools)
+    if terminal_tools:
+        additional_tools.extend(terminal_tools)
+    if planning_tools:
+        additional_tools.extend(planning_tools)
     if mcp_tools:
         additional_tools.extend(mcp_tools)
     

alita_sdk/cli/agent_loader.py

@@ -5,13 +5,42 @@ Handles loading agent definitions from various file formats (YAML, JSON, Markdow
 """
 
 import json
+import locale
 import yaml
 from pathlib import Path
 from typing import Dict, Any
+from pydantic import SecretStr
 
 from .config import substitute_env_vars
 
 
+def _read_text_with_fallbacks(path: Path) -> str:
+    """Read a text file using robust, cross-platform defaults.
+
+    Why this exists:
+    - On Windows, `Path.read_text()` defaults to the system code page (often cp1252).
+      Agent definition files are commonly authored as UTF-8 and may include smart quotes
+      (e.g. ”) whose UTF-8 byte sequence contains 0x9D, which is *undefined* in cp1252.
+      That combination triggers: UnicodeDecodeError: 'charmap' codec can't decode byte 0x9d.
+    """
+
+    # 1) Prefer UTF-8 (most common for repo files)
+    try:
+        return path.read_text(encoding="utf-8")
+    except UnicodeDecodeError:
+        pass
+
+    # 2) UTF-8 with BOM (common on Windows)
+    try:
+        return path.read_text(encoding="utf-8-sig")
+    except UnicodeDecodeError:
+        pass
+
+    # 3) Fall back to the platform preferred encoding, but never crash.
+    #    This keeps the CLI usable even if a file was authored in a legacy encoding.
+    return path.read_text(encoding=locale.getpreferredencoding(False), errors="replace")
+
+
 def load_agent_definition(file_path: str) -> Dict[str, Any]:
     """
     Load agent definition from file.
@@ -31,8 +60,8 @@ def load_agent_definition(file_path: str) -> Dict[str, Any]:
     
     if not path.exists():
         raise FileNotFoundError(f"Agent definition not found: {file_path}")
-    
-    content = path.read_text()
+
+    content = _read_text_with_fallbacks(path)
     
     # Handle markdown with YAML frontmatter
     if path.suffix == '.md':
@@ -57,7 +86,8 @@ def load_agent_definition(file_path: str) -> Dict[str, Any]:
                     'filesystem_tools_preset': frontmatter.get('filesystem_tools_preset'),
                     'filesystem_tools_include': frontmatter.get('filesystem_tools_include'),
                     'filesystem_tools_exclude': frontmatter.get('filesystem_tools_exclude'),
-                    'mcps': frontmatter.get('mcps', [])
+                    'mcps': frontmatter.get('mcps', []),
+                    'persona': frontmatter.get('persona')
                 }
         
         # Plain markdown - use content as system prompt
@@ -85,6 +115,25 @@ def load_agent_definition(file_path: str) -> Dict[str, Any]:
     raise ValueError(f"Unsupported file format: {path.suffix}")
 
 
+def unwrap_secrets(obj: Any) -> Any:
+    """
+    Recursively unwrap pydantic SecretStr values into plain strings.
+
+    Handles nested dicts, lists, tuples, and sets while preserving structure.
+    """
+    if isinstance(obj, SecretStr):
+        return obj.get_secret_value()
+    if isinstance(obj, dict):
+        return {k: unwrap_secrets(v) for k, v in obj.items()}
+    if isinstance(obj, list):
+        return [unwrap_secrets(v) for v in obj]
+    if isinstance(obj, tuple):
+        return tuple(unwrap_secrets(v) for v in obj)
+    if isinstance(obj, set):
+        return {unwrap_secrets(v) for v in obj}
+    return obj
+
+
 def build_agent_data_structure(agent_def: Dict[str, Any], toolkit_configs: list, 
                                llm_model: str, llm_temperature: float, llm_max_tokens: int) -> Dict[str, Any]:
     """
@@ -128,7 +177,8 @@ def build_agent_data_structure(agent_def: Dict[str, Any], toolkit_configs: list,
                     if hasattr(toolkit_class, 'toolkit_config_schema'):
                         schema = toolkit_class.toolkit_config_schema()
                         validated_config = schema(**toolkit_config)
-                        validated_dict = validated_config.model_dump()
+                        # Use python mode so SecretStr remains as objects, then unwrap recursively
+                        validated_dict = unwrap_secrets(validated_config.model_dump(mode="python"))
                         validated_dict['type'] = toolkit_config.get('type')
                         validated_dict['toolkit_name'] = toolkit_config.get('toolkit_name')
                         validated_toolkit_configs.append(validated_dict)
@@ -168,7 +218,6 @@ def build_agent_data_structure(agent_def: Dict[str, Any], toolkit_configs: list,
                 'settings': toolkit_config,
                 'selected_tools': toolkit_config.get('selected_tools', [])
             })
-    
     return {
         'instructions': agent_def.get('system_prompt', ''),
         'tools': tools,
@@ -181,8 +230,6 @@ def build_agent_data_structure(agent_def: Dict[str, Any], toolkit_configs: list,
             'model_name': llm_model,
             'max_tokens': llm_max_tokens,
             'temperature': llm_temperature,
-            'top_p': 1.0,
-            'top_k': 0,
             'integration_uid': None,
             'indexer_config': {
                 'ai_model': 'langchain_openai.ChatOpenAI',
@@ -193,5 +240,6 @@ def build_agent_data_structure(agent_def: Dict[str, Any], toolkit_configs: list,
                 }
             }
         },
-        'agent_type': agent_def.get('agent_type', 'react')
+        'agent_type': agent_def.get('agent_type', 'react'),
+        'persona': agent_def.get('persona', 'quirky')
     }

alita_sdk/cli/agent_ui.py

@@ -79,12 +79,23 @@ def print_help():
         padding=(0, 1),
     )
     
-    table.add_column("Command", style="bold yellow", no_wrap=True, width=12)
+    table.add_column("Command", style="bold yellow", no_wrap=True, width=16)
     table.add_column("Description", style="white")
     
     table.add_row("/clear", "Clear conversation history")
     table.add_row("/history", "Show conversation history")
     table.add_row("/save", "Save conversation to file")
+    table.add_row("/agent", "Switch to a different agent or direct chat")
+    table.add_row("/model", "Switch to a different model (preserves history)")
+    table.add_row("/reload", "Reload agent from file (hot reload)")
+    table.add_row("/mode", "Set approval mode: always, auto, yolo")
+    table.add_row("/dir [add|rm] <path>", "Add/remove/list allowed directories")
+    table.add_row("/inventory <path>", "Load inventory/knowledge graph from JSON file")
+    table.add_row("/session", "List or resume previous sessions with plans")
+    table.add_row("/add_mcp", "Add an MCP server (preserves history)")
+    table.add_row("/add_toolkit", "Add a toolkit (preserves history)")
+    table.add_row("/rm_mcp", "Remove an MCP server")
+    table.add_row("/rm_toolkit", "Remove a toolkit")
     table.add_row("/help", "Show this help")
     table.add_row("exit", "End conversation")
     
@@ -92,55 +103,106 @@ def print_help():
     console.print()
 
 
-def print_welcome(agent_name: str, agent_type: str = "Local Agent"):
-    """Print combined welcome banner with logo, agent info, and help."""
+def print_welcome(agent_name: str, model: str = "gpt-4o", temperature: float = 0.1, mode: str = "always"):
+    """Print combined welcome banner with logo, agent info, and help in two columns."""
     
     version = get_version()
     
-    # Build the complete welcome message using Text objects
-    content = Text()
+    # Mode display with color
+    mode_colors = {'always': 'yellow', 'auto': 'green', 'yolo': 'red'}
+    mode_color = mode_colors.get(mode, 'white')
     
-    # Add logo lines with cyan styling
+    # Left column: Logo + version + agent info
+    left_content = Text()
     for line in ALITA_LOGO:
-        content.append("    " + line + "\n", style="bold cyan")
-    
-    content.append("                    CLI ", style="dim")
-    content.append(f"v{version}\n\n", style="bold white")
-    
-    # Connection status
-    content.append("    ● ", style="bold green")
-    content.append("Agent: ", style="bold white")
-    content.append(f"{agent_name}\n", style="bold cyan")
-    content.append("    ● ", style="bold green")
-    content.append("Type:  ", style="bold white")
-    content.append(f"{agent_type}\n\n", style="cyan")
-    
-    # Quick help section
-    content.append("    Type a message to chat, or use these commands:\n\n", style="dim")
-    
-    # Commands
-    content.append("    /help", style="bold yellow")
-    content.append("    Show all commands\n", style="white")
-    content.append("    /clear", style="bold yellow")
-    content.append("   Clear history\n", style="white")
-    content.append("    exit", style="bold yellow")
-    content.append("     End conversation\n", style="white")
+        left_content.append(line + "\n", style="bold cyan")
+    left_content.append("        CLI ", style="dim")
+    left_content.append(f"v{version}\n\n", style="bold white")
+    left_content.append("● ", style="bold green")
+    left_content.append("Agent: ", style="bold white")
+    left_content.append(f"{agent_name}\n", style="bold cyan")
+    left_content.append("● ", style="bold green")
+    left_content.append("Model: ", style="bold white")
+    left_content.append(f"{model}", style="cyan")
+    left_content.append(" | ", style="dim")
+    left_content.append("Temp: ", style="bold white")
+    left_content.append(f"{temperature}\n", style="cyan")
+    left_content.append("● ", style="bold green")
+    left_content.append("Mode:  ", style="bold white")
+    left_content.append(f"{mode}\n", style=f"bold {mode_color}")
+    
+    # Right column: Commands
+    right_content = Text()
+    right_content.append("\n")  # Align with logo
+    right_content.append("/help", style="bold yellow")
+    right_content.append("        Show all commands\n", style="dim")
+    right_content.append("/agent", style="bold yellow")
+    right_content.append("       Switch agent\n", style="dim")
+    right_content.append("/model", style="bold yellow")
+    right_content.append("       Switch model\n", style="dim")
+    right_content.append("/reload", style="bold yellow")
+    right_content.append("      Reload agent file\n", style="dim")
+    right_content.append("/mode", style="bold yellow")
+    right_content.append("        Set approval mode\n", style="dim")
+    right_content.append("/dir", style="bold yellow")
+    right_content.append("         Add/list directories\n", style="dim")
+    right_content.append("/session", style="bold yellow")
+    right_content.append("     List/resume sessions\n", style="dim")
+    right_content.append("/inventory", style="bold yellow")
+    right_content.append("   Load knowledge graph\n", style="dim")
+    right_content.append("/add_mcp", style="bold yellow")
+    right_content.append("     Add MCP server\n", style="dim")
+    right_content.append("/add_toolkit", style="bold yellow")
+    right_content.append(" Add toolkit\n", style="dim")
+    right_content.append("exit", style="bold yellow")
+    right_content.append("         End conversation\n", style="dim")
+    
+    # Create two-column layout
+    columns = Columns([left_content, right_content], padding=(0, 4), expand=False)
     
     console.print()
     console.print(Panel(
-        content,
+        columns,
         box=box.DOUBLE,
         border_style="cyan",
-        padding=(0, 2),
+        padding=(1, 2),
     ))
     console.print()
 
 
+def render_plan(plan_state) -> None:
+    """Render a plan with checkboxes."""
+    if not plan_state or not plan_state.steps:
+        return
+    
+    console.print(plan_state.render())
+    console.print()
+
+
+def render_approval_prompt(tool_name: str, tool_args: dict, truncate: int = 60) -> str:
+    """
+    Render an approval prompt for a tool call.
+    
+    Returns the formatted string for display in the input box.
+    """
+    # Format args as a compact string
+    args_str = ", ".join(f"{k}={repr(v)[:truncate]}" for k, v in tool_args.items())
+    if len(args_str) > 80:
+        args_str = args_str[:77] + "..."
+    
+    return f"🔧 {tool_name}: {args_str}"
+
+
 def display_output(agent_name: str, message: str, output: str):
     """Display agent output with markdown rendering if applicable."""
     console.print(f"\n[bold cyan]🤖 Agent: {agent_name}[/bold cyan]\n")
     console.print(f"[bold]Message:[/bold] {message}\n")
     console.print("[bold]Response:[/bold]")
+    
+    # Ensure output is a string
+    if not isinstance(output, str):
+        output = str(output)
+    
     if any(marker in output for marker in ['```', '**', '##', '- ', '* ']):
         console.print(Markdown(output))
     else: